from torch.utils.data import DataLoader


class StockDataLoader(DataLoader):

    def __init__(self, dataset, batch_size, shuffle=True):
        self.dataset = dataset
        super().__init__(self.dataset, batch_size=batch_size, shuffle=shuffle)
